from langchain.chains.llm import LLMChain
from langchain_community.llms.tongyi import Tongyi
from langchain_core.prompts import PromptTemplate

template_multiple_choice = """请针对>>>和<<<中间的用户问题，选择一个合适的工具去回答它的问题，只要用A、B、C的选项字母告诉我答案，如果你觉得不合适，请选D.
>>>{question}<<<
我们有的工具包括：
A、一个能够查询商品信息，为用户进行商品导航的工具
B、一个能够查询订单信息，获取最新的订单情况的工具
C、一个能够搜索商家的退换货政策、运费、物流时长、支付渠道、覆盖国家的工具
D、都不合适
"""
prompt_template = PromptTemplate(template=template_multiple_choice)

chain = LLMChain(
    prompt=prompt_template,
    llm=Tongyi(),
    verbose=True,
)
question = "我想买一件衣服，有推荐的吗？？？"

ret = chain(question)


def goods():
    return "返回推荐的商品"


if ret == "A":
    goods()

print(ret["text"])
